solve bug
